package com.just.servlet;

import com.sun.xml.internal.bind.v2.schemagen.XmlSchemaGenerator;
import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.ProgressListener;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.util.List;
import java.util.UUID;

public class FileUpServlet extends HttpServlet {

    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {

    }

    @Override
    protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        //判断上传的文件是普通表单，还是带文件的表单
        if (!ServletFileUpload.isMultipartContent(req)) {
            return; //如果是普通的表单终止方法运行，直接返回
        }//通过if则说明是上传文件


        //创建上传文件的保存路径。建议放在web——IFN路径下，安全，用户无法直接访问上传的文件
        String upLoadPath = this.getServletContext().getRealPath("/WEB-INF/upload");
        System.out.println("upLoadPath:"+upLoadPath);
        File upLoadfile = new File(upLoadPath);
        if (!upLoadfile.exists()) {
            //System.out.println("创建/WEB-INF/upload");
            upLoadfile.mkdirs();//创建这个目录
        }

        //缓存，临时文件
        //临时路径，假如文件超过了预期大小，我们就把他放在一个临时文件中，过几天自动伤处，或者提醒用户保存问永久
        String tempPath = this.getServletContext().getRealPath("/WEB-INF/tmp");
        System.out.println("tempPath:"+tempPath);
        File tempFile = new File(tempPath);
        if (!tempFile.exists()) {
            //System.out.println("创建/WEB-INF/tmp");
            tempFile.mkdirs();//创建这个临时目录
        }

        //处理上传文件，一般需要通过流来获取，我们可以使用request.getInputStream(),获取原生太的文件上传流，十分麻烦
        //可以使用Apache的文件上传组件来实现，common-fileupLoad，他需要依赖common-io
        try {
            //1、创建DiskFileItemFactory对象，处理文件上传路径或者大小限制的
            DiskFileItemFactory factory = getDiskFileItemFactory(tempFile);
            //2、获取servletFileLoad
            ServletFileUpload upload = getServletFileUpLoad(factory);
            //3、处理上传文件
            String msg = upLoadParseRequest(upload, req, upLoadPath);
            //servlet请求转发消息
            req.setAttribute("msg", msg);
            req.getRequestDispatcher("info.jsp").forward(req, resp);

        } catch (FileUploadException e) {
            e.printStackTrace();
        }
    }


    public static DiskFileItemFactory getDiskFileItemFactory(File tempFile) {
        DiskFileItemFactory factory = new DiskFileItemFactory();
        //通过这个工厂设置一个缓冲区，当上传文件大于这个缓冲区的时候，将他放在临时文件中
        factory.setSizeThreshold(1024 * 1024*10); //缓存大小为1M
        factory.setRepository(tempFile);//临时文件存放的目录，需要一个file
        return factory;
    }

    public static ServletFileUpload getServletFileUpLoad(DiskFileItemFactory factory) {
        //2、获取servletFileLoad
        ServletFileUpload upload = new ServletFileUpload(factory);
        //监听文件上传进度
        upload.setProgressListener(new ProgressListener() {
            //pBytesRead:以及读取到文件的大小
            //pContentLength:文件大小
            @Override
            public void update(long pBytesRead, long pContentLength, int pItem) {
                System.out.println("总大小:" + pContentLength + " 已经上传:" + pBytesRead);
            }
        });
        //处理乱码
        upload.setHeaderEncoding("UTF-8");
        //设置单个文件的最大值
        upload.setFileSizeMax(1024 * 1024 * 10);
        //设置总共能够上传文件的大小
        upload.setSizeMax(1024 * 1024 * 10);
        return upload;
    }

    public static String upLoadParseRequest(ServletFileUpload upload, HttpServletRequest request, String upLoadPath) throws FileUploadException, IOException {
        String msg = "";
        //3、处理上传文件
        //把前端请求，封装成一个FileItem对象,需要从ServletFileUpload对象中获取
        List<FileItem> fileItems = upload.parseRequest(request);
        //fileItem是每一个form表单中的元素，区分哪个带不带文件
        for (FileItem fileItem : fileItems) {
            //判断上传的文件是普通表单，还是带文件的表单
            if (fileItem.isFormField()) {//是普通表单的情况下
                //getFieldName指的是前端表单控件的name
                String name = fileItem.getFieldName();
                String value = fileItem.getString("UTF-8");//处理乱码
                System.out.println(name + ":" + value);
            } else {
                //是含有文件的表单的情况下
                //=========================处理文件================================================================//
                String upLoadFileName = fileItem.getName();
                System.out.println("上传的文件名"+ upLoadFileName);
                //可能存在文件名不合法的存在
                if (upLoadFileName.trim().equals("") || upLoadFileName == null) {
                    continue;
                }
                //获得上传的文件名 /images/girl/1.jpg
                String fileName = upLoadFileName.substring(upLoadFileName.lastIndexOf("/") + 1);
                //获取文件的后缀名
                String fileExtName = upLoadFileName.substring(upLoadFileName.lastIndexOf(".") + 1);
                System.out.println("文件信息 【文件名:" + fileName + "  / 文件类型:"+fileExtName );

                //可以使用uuid(唯一识别的通用码) 保证文件名唯一
                //uuid.randomUUID(),随机生成一个唯一识别通用码
                //网络传输中的东西，都需要序列化
                //pojo ,实体类，如果想要在多个电脑上运行，传输===》需要把对象序列化
                //implements java.io.Serializable  （标记接口） JVM--->java栈       本地方法栈 native ---》 C++
                // （jni == java Native Interface）
                String uuidPath = UUID.randomUUID().toString();

                //=========================存放地址=================================================================//

                //存在哪里 uploadPath
                //文件真实存在的路径 realPath
                String realPath = upLoadPath + "/" + uuidPath;
                //给每个文件创建一个对应的文件夹
                File realPathFile = new File(realPath);
                if (!realPathFile.exists()) {
                    realPathFile.mkdirs();
                }

                //=========================文件传输==================================================================//
                //获取文件上传的流
                InputStream inputStream = fileItem.getInputStream();

                //创建一个文件输出流
                //realPath = 真实的文件夹
                //差了一个文件；加上输出文件的名字 + “/” + uuidFileName
                FileOutputStream fos = new FileOutputStream(realPath + "/" + fileName);

                //创建一个缓冲区
                byte[] buffer = new byte[1024 * 1024];

                //判断是否读取完毕
                int len = 0;
                while ((len = inputStream.read(buffer)) > 0) {
                    fos.write(buffer, 0, len);
                }
                //关闭流
                fos.close();
                inputStream.close();

                msg = "文件上传成功";
                fileItem.delete();//上传成功，清楚零时文件
            }
        }
        return msg;
    }


}
